package br.com.academia.DAO;

import java.sql.Connection;
import java.sql.PreparedStatement;
import java.sql.ResultSet;
import java.sql.SQLException;
import java.util.ArrayList;
import java.util.List;

import br.com.academia.beans.Treino;

public class TreinoDAO {
	
	private Connection connection;

	public TreinoDAO(Connection connection) {
		this.connection = connection;
	}
	
	public List<String[]> consultaTreinos(List<String> dadosConsulta)  {
		List<String[]> retorno = new ArrayList<String[]>();
		String query = "Select idTreino, Aluno_idAluno, Titulo, Descricao " +
				"from Treinos ";// +
				//"Left Join Alunos on Treinos.Empresa_idEmpresa = Empresas.idEmpresa";
		 try {
			 PreparedStatement stm = connection.prepareStatement(query);
			 ResultSet rs = stm.executeQuery();
			 while(rs.next()) {
				 String[] row = new String[dadosConsulta.size()];
				 for(int i = 0; i < row.length; i++) {
					 row[i] = rs.getString(dadosConsulta.get(i));
				 }
				 retorno.add(row);
			 }
		 } catch(SQLException e) {
			 e.printStackTrace();
		 }
		return retorno;
	}
	
	public Treino consultaTreinoId(int idTreino) {
		Treino treino = new Treino();
		String query = "Select * from Treinos where idTreino = ?";
		try {
			PreparedStatement stm = connection.prepareStatement(query);
			stm.setInt(1, idTreino);
			ResultSet rs = stm.executeQuery();
			
			if(rs.next()) {
				treino.setIdTreino(rs.getInt("idTreino"));
				treino.setTitulo(rs.getString("titulo"));
				treino.setDescricao(rs.getString("descricao"));
			}
		}catch(SQLException e) {
			e.printStackTrace();
		}
		
		return treino;
	}
	
	public List<Treino> buscaTreinoSelecao() {
		List<Treino> listaTreino = new ArrayList<Treino>();
		String query = "Select idTreino, titulo, Aluno_idAluno from Treinos";

		try {
			PreparedStatement stm = connection.prepareStatement(query);
			ResultSet rs = stm.executeQuery();

			while (rs.next()) {
				Treino treino = new Treino();
				treino.setIdTreino(rs.getInt("idTreino"));
				treino.setTitulo(rs.getString("titulo"));
				treino.setAluno_idAluno(rs.getInt("aluno_idAluno"));
				listaTreino.add(treino);
			}
		} catch (SQLException e) {
			e.printStackTrace();
		}
		
		return listaTreino;
	}
	
	public List<Treino> buscaTreinoSelecao(int idAluno) {
		List<Treino> listaTreino = new ArrayList<Treino>();
		String query = "Select idTreino, titulo, Aluno_idAluno from Treinos where Aluno_idAluno = ?";
		
		try {
			PreparedStatement stm = connection.prepareStatement(query);
			stm.setInt(1, idAluno);
			ResultSet rs = stm.executeQuery();

			while (rs.next()) {
				Treino treino = new Treino();
				treino.setIdTreino(rs.getInt("idTreino"));
				treino.setTitulo(rs.getString("titulo"));
				treino.setAluno_idAluno(rs.getInt("aluno_idAluno"));
				listaTreino.add(treino);
			}
		} catch (SQLException e) {
			e.printStackTrace();
		}
		
		return listaTreino;
	}
	
	public List<Treino> buscaTreinosIdAluno(int idAluno) {
		List<Treino> listaTreino = new ArrayList<Treino>();
		String query = "Select idTreino, titulo from Treinos where aluno_idAluno = ?";

		try {
			PreparedStatement stm = connection.prepareStatement(query);
			stm.setInt(1, idAluno);
			ResultSet rs = stm.executeQuery();

			while (rs.next()) {
				Treino treino = new Treino();
				treino.setIdTreino(rs.getInt("idTreino"));
				treino.setTitulo(rs.getString("titulo"));
				listaTreino.add(treino);
			}
		} catch (SQLException e) {
			e.printStackTrace();
		}
		
		return listaTreino;

	}
	
}
